package com.gitee.jdbc.elasticsearch;

import java.sql.Connection;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.SQLWarning;
import java.sql.Statement;
import java.util.LinkedHashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;

public class ElasticsearchStatement implements Statement {

  private final ElasticsearchConnection connection;

  public ElasticsearchStatement(ElasticsearchConnection connection) {
    this.connection = connection;
  }

  @Override
  public ResultSet executeQuery(String sql) throws SQLException {
    String indexName = sql;

    ElasticsearchClient client = connection.getClient();
    List<Map> result = client.queryAll(indexName);
    Set<String> columns = new LinkedHashSet<>();
    for (Map row : result) {
      for (Object key : row.keySet()) {
        if (!columns.contains(key.toString())) {
          columns.add(key.toString());
        }
      }
    }
    ElasticsearchResultSet resultSet = new ElasticsearchResultSet();
    resultSet.setColumnNames(columns.toArray(new String[0]));
    for (Map row : result) {
      String[] columnValues = new String[columns.size()];
      int i = 0;
      for (String key : columns) {
        Object v = row.get(key);
        columnValues[i++] = (null == v) ? null : v.toString();
      }
      resultSet.addRow(columnValues);
    }
    return resultSet;
  }

  private int executeInsert(String sql) {
    ElasticsearchClient client = connection.getClient();
    WriteStatementEntity entity = JsonUtils.toBeanObject(sql, WriteStatementEntity.class);
    client.bulkInsertDocuments(entity.getIndexName(), entity.getSources());
    return entity.getSources().size();
  }

  @Override
  public int executeUpdate(String sql) throws SQLException {
    return executeInsert(sql);
  }

  @Override
  public void close() throws SQLException {

  }

  @Override
  public int getMaxFieldSize() throws SQLException {
    return 0;
  }

  @Override
  public void setMaxFieldSize(int max) throws SQLException {

  }

  @Override
  public int getMaxRows() throws SQLException {
    return 0;
  }

  @Override
  public void setMaxRows(int max) throws SQLException {

  }

  @Override
  public void setEscapeProcessing(boolean enable) throws SQLException {

  }

  @Override
  public int getQueryTimeout() throws SQLException {
    return 0;
  }

  @Override
  public void setQueryTimeout(int seconds) throws SQLException {

  }

  @Override
  public void cancel() throws SQLException {

  }

  @Override
  public SQLWarning getWarnings() throws SQLException {
    return null;
  }

  @Override
  public void clearWarnings() throws SQLException {

  }

  @Override
  public void setCursorName(String name) throws SQLException {

  }

  @Override
  public boolean execute(String sql) throws SQLException {
    return executeInsert(sql) > 0;
  }

  @Override
  public ResultSet getResultSet() throws SQLException {
    return null;
  }

  @Override
  public int getUpdateCount() throws SQLException {
    return 0;
  }

  @Override
  public boolean getMoreResults() throws SQLException {
    return false;
  }

  @Override
  public void setFetchDirection(int direction) throws SQLException {

  }

  @Override
  public int getFetchDirection() throws SQLException {
    return 0;
  }

  @Override
  public void setFetchSize(int rows) throws SQLException {

  }

  @Override
  public int getFetchSize() throws SQLException {
    return 0;
  }

  @Override
  public int getResultSetConcurrency() throws SQLException {
    return 0;
  }

  @Override
  public int getResultSetType() throws SQLException {
    return 0;
  }

  @Override
  public void addBatch(String sql) throws SQLException {

  }

  @Override
  public void clearBatch() throws SQLException {

  }

  @Override
  public int[] executeBatch() throws SQLException {
    return new int[0];
  }

  @Override
  public Connection getConnection() throws SQLException {
    return connection;
  }

  @Override
  public boolean getMoreResults(int current) throws SQLException {
    return false;
  }

  @Override
  public ResultSet getGeneratedKeys() throws SQLException {
    return null;
  }

  @Override
  public int executeUpdate(String sql, int autoGeneratedKeys) throws SQLException {
    return executeInsert(sql);
  }

  @Override
  public int executeUpdate(String sql, int[] columnIndexes) throws SQLException {
    return executeInsert(sql);
  }

  @Override
  public int executeUpdate(String sql, String[] columnNames) throws SQLException {
    return executeInsert(sql);
  }

  @Override
  public boolean execute(String sql, int autoGeneratedKeys) throws SQLException {
    return executeInsert(sql) > 0;
  }

  @Override
  public boolean execute(String sql, int[] columnIndexes) throws SQLException {
    return executeInsert(sql) > 0;
  }

  @Override
  public boolean execute(String sql, String[] columnNames) throws SQLException {
    return executeInsert(sql) > 0;
  }

  @Override
  public int getResultSetHoldability() throws SQLException {
    return 0;
  }

  @Override
  public boolean isClosed() throws SQLException {
    return false;
  }

  @Override
  public void setPoolable(boolean poolable) throws SQLException {

  }

  @Override
  public boolean isPoolable() throws SQLException {
    return false;
  }

  @Override
  public void closeOnCompletion() throws SQLException {

  }

  @Override
  public boolean isCloseOnCompletion() throws SQLException {
    return false;
  }

  @Override
  public <T> T unwrap(Class<T> iface) throws SQLException {
    return null;
  }

  @Override
  public boolean isWrapperFor(Class<?> iface) throws SQLException {
    return false;
  }
}
